package cn.lichenfei.app.satoken.config;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.context.model.SaResponse;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * [Sa-Token 权限认证] 配置类：过滤器方式 (使用过滤器的话注解会不起作用)
 */
@Slf4j
//@Configuration
public class SaTokenFilterConfigure {

    /**
     * 注册 [Sa-Token全局过滤器]
     */
    @Bean
    public SaServletFilter getSaServletFilter() {
        return new SaServletFilter()

                // 指定 拦截路由 与 放行路由
                .addInclude("/**")
                /* 排除掉 /favicon.ico swagger*/
                .addExclude("/favicon.ico", "/swagger-ui.html", "/swagger-resources/**", "/webjars/**", "/doc.html", "/v2/api-docs")

                // 认证函数: 每次请求执行
                .setAuth(obj -> {
                    System.out.println("---------- 进入Sa-Token全局认证 -----------");
                    String method = SaHolder.getRequest().getMethod();

                    String url = SaHolder.getRequest().getRequestPath();
                    SaRouter
                            //拦截所有路由
                            .match("/**")
                            //排除/user/doLogin 用于开放登录
                            .notMatch("/user/doLogin")
                            .check(() -> StpUtil.checkLogin())
                            //权限效验
                            .match(url, () -> StpUtil.checkPermission("user:add"));
                })
                // 异常处理函数：每次认证函数发生异常时执行此函数
                .setError(e -> {
                    System.out.println("---------- 进入Sa-Token异常处理 -----------");
                    SaResponse response = SaHolder.getResponse();
                    response.setHeader("Content-Type", "application/json;charset=UTF-8");
                    return SaResult.error(e.getMessage());
                })
                // 前置函数：在每次认证函数之前执行
                .setBeforeAuth(r -> {
                    // ---------- 设置一些安全响应头 ----------
                    SaHolder.getResponse()
                            // 服务器名称
                            .setServer("sa-server")
                            // 是否可以在iframe显示视图： DENY=不可以 | SAMEORIGIN=同域下可以 | ALLOW-FROM uri=指定域名下可以
                            .setHeader("X-Frame-Options", "SAMEORIGIN")
                            // 是否启用浏览器默认XSS防护： 0=禁用 | 1=启用 | 1; mode=block 启用, 并在检查到XSS攻击时，停止渲染页面
                            .setHeader("X-XSS-Protection", "1; mode=block")
                            // 禁用浏览器内容嗅探
                            .setHeader("X-Content-Type-Options", "nosniff")
                    ;
                })
                ;
    }
}
